Build: Test the API

# tests/test_api.py — tests for the FastAPI endpoints.
# FastAPI's TestClient simulates HTTP requests without a real server.
# Run with: pytest tests/test_api.py -v
import pytest
from fastapi.testclient import TestClient
from api import app, db

# The TestClient wraps your app and lets you make requests.
# It handles startup and shutdown events automatically.
client = TestClient(app)


# --- GET /prompts ---

def test_list_prompts_returns_list():
    """GET /prompts should return a JSON array."""
    response = client.get("/prompts")
    assert response.status_code == 200
    assert isinstance(response.json(), list)


def test_list_prompts_with_platform_filter():
    """GET /prompts?platform=Kling should only return Kling prompts."""
    # First add a known prompt so we have something to filter on
    data = {"platform": "Kling", "shot": "Filter test shot", "prompt_text": "Filter test prompt"}
    client.post("/prompts", json=data)

    response = client.get("/prompts", params={"platform": "Kling"})
    assert response.status_code == 200
    results = response.json()
    # Every result should be a Kling prompt
    for r in results:
        assert r["platform"] == "Kling"


# --- POST /prompts ---

def test_create_prompt_success():
    """POST /prompts with valid data should return 201 and the created prompt."""
    data = {
        "platform": "Runway",
        "shot": "Test shot from pytest",
        "prompt_text": "Cinematic test prompt for API test"
    }
    response = client.post("/prompts", json=data)

    assert response.status_code == 201
    result = response.json()
    assert result["platform"] == "Runway"
    assert result["shot"] == "Test shot from pytest"
    assert "id" in result  # database assigned an ID
    assert "created_at" in result


def test_create_prompt_invalid_platform():
    """POST /prompts with unrecognized platform should return 422."""
    data = {
        "platform": "Midjourney",  # not a valid platform
        "shot": "test",
        "prompt_text": "test"
    }
    response = client.post("/prompts", json=data)
    # 422 = Unprocessable Entity — Pydantic validation failed
    assert response.status_code == 422


def test_create_prompt_empty_shot():
    """POST /prompts with empty shot should return 422."""
    data = {
        "platform": "Kling",
        "shot": "   ",  # whitespace only
        "prompt_text": "test"
    }
    response = client.post("/prompts", json=data)
    assert response.status_code == 422


# --- GET /prompts/{id} ---

def test_get_prompt_by_id():
    """GET /prompts/{id} should return the prompt if it exists."""
    # Create one first
    data = {"platform": "Veo", "shot": "Get-by-id test", "prompt_text": "Get-by-id prompt"}
    create = client.post("/prompts", json=data)
    prompt_id = create.json()["id"]

    # Now fetch it by ID
    response = client.get(f"/prompts/{prompt_id}")
    assert response.status_code == 200
    assert response.json()["id"] == prompt_id
    assert response.json()["platform"] == "Veo"


def test_get_prompt_not_found():
    """GET /prompts/99999 for a non-existent ID should return 404."""
    response = client.get("/prompts/99999")
    assert response.status_code == 404


# --- DELETE /prompts/{id} ---

def test_delete_prompt():
    """DELETE /prompts/{id} should remove the prompt and return confirmation."""
    # Create one to delete
    data = {"platform": "Kling", "shot": "Delete me", "prompt_text": "To be deleted"}
    create = client.post("/prompts", json=data)
    prompt_id = create.json()["id"]

    # Delete it
    response = client.delete(f"/prompts/{prompt_id}")
    assert response.status_code == 200

    # Verify it's gone
    get_response = client.get(f"/prompts/{prompt_id}")
    assert get_response.status_code == 404


def test_delete_nonexistent_prompt():
    """DELETE /prompts/99999 should return 404."""
    response = client.delete("/prompts/99999")
    assert response.status_code == 404


# --- GET /stats ---

def test_get_stats():
    """GET /stats should return a dictionary mapping platform names to counts."""
    response = client.get("/stats")
    assert response.status_code == 200
    stats = response.json()
    assert isinstance(stats, dict)
    # All values should be integers
    for platform, count in stats.items():
        assert isinstance(count, int)
        assert count > 0